package com.hongawen.mybatis.utils;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;

import java.io.File;
import java.util.Collections;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * @author hongawen
 * @version 1.0.0
 * @date 2021年12月13日 21:01
 */
public class AutoGenerateCode {


    private static final String TARGET_DIR = "D://code";

    private static final String DB_URL = "jdbc:mysql://47.97.75.166:3306/shiro";

    private static final String USERNAME = "root";

    private static final String PASSWORD = "123456";


    public static void main(String[] args) {
        //文件夹提前建好，否则会报错
        FastAutoGenerator.create(DB_URL, USERNAME, PASSWORD)
                .globalConfig(builder -> {
                    builder.author("hongawen") // 设置作者
                            .fileOverride() // 覆盖已生成文件
                            .outputDir(TARGET_DIR); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.hongawen") // 设置父包名
                            .moduleName("shiro") // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, TARGET_DIR + File.separator + "shiro")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.addInclude("user","permission","role","role_permission","user_role") // 设置需要生成的表名
                            .addTablePrefix("t_")// 设置过滤表前缀
                            .entityBuilder()
//                            .superClass(com.njcn.db.bo.BaseEntity.class) //默认实体继承指定父类
                            .enableLombok() //启用lombok
                            .enableRemoveIsPrefix() //开启 Boolean 类型字段移除 is 前缀
                            .controllerBuilder()
//                            .superClass(com.njcn.web.controller.BaseController.class) //默认controller继承指定父类
                            .enableRestStyle() //开启生成@RestController 控制器
                    ;
                })
                .execute();
    }


}
